package demo3;

import java.util.concurrent.BlockingQueue;
import java.util.concurrent.PriorityBlockingQueue;

/**
 * 自定义定时器
 */

public class MyTimer {
    // 用一个阻塞队列来组织任务
    private BlockingQueue<MyTask> queue = new PriorityBlockingQueue<>();

    public MyTimer () {
        // 创建扫描线程
        Thread thread = new Thread(() -> {
           // 不断地扫描线程队列中的任务
           while (true) {
               try {
                   // 1. 从队列中取出任务
                   MyTask task = queue.take();
                   // 2. 判断到没到执行时间
                   long currentTime = System.currentTimeMillis();
                   if (currentTime >= task.getTime()) {
                       // 时间到了, 执行任务
                       task.getRunnable().run();
                   } else {
                       // 没有到时间, 重新放回队列
                       queue.put(task);
                   }
               } catch (InterruptedException e) {
                   e.printStackTrace();
               }
           }
        });
        // 启动线程
        thread.start();
    }

    /**
     * 添加任务的方法
     * @param runnable 任务
     * @param delay 延时
     * @throws InterruptedException
     */
    public void schedule (Runnable runnable, long delay) throws InterruptedException {
        // 根据传入的参数, 构造一个MyTask
        MyTask task = new MyTask(runnable, delay);
        // 把任务放入阻塞队列
        queue.put(task);
    }
}

// 用一个类来描述任务及任务执行的时间
class MyTask implements Comparable<MyTask> {
    // 任务
    private Runnable runnable;
    // 任务执行的时间
    private long time;

    public MyTask (Runnable runnable, long time) {
        // 校验任务不能为空
        if (runnable == null) {
            throw new IllegalArgumentException("任务不能为空...");
        }
        // 校验时间不能为负数
        if (time < 0) {
            throw new RuntimeException("时间不能为负数...");
        }
        this.runnable = runnable;
        this.time = time + System.currentTimeMillis();
    }

    public Runnable getRunnable () {
        return runnable;
    }

    public long getTime () {
        return time;
    }

    @Override
    public int compareTo(MyTask o) {
        // 返回小根堆
//        return (int) (this.time- o.getTime());
        if (this.getTime() > o.getTime()) return 1;
        else if (this.getTime() < o.getTime()) return -1;
        else return 0;
    }
}